# 该作业用来计算某个目录的大小，目录下面可能还有子目录
import os

# 自定义计算目录大小函数
def calsize(dir1):
	#获取被目录中的所有文件信息
	dlist = os.listdir(dir1)
	#遍历所有文件，计算大小
	wj = []# 存放目录子目录或文件大小的列表
	for f in dlist:
		#为遍历的文件添加目录路径
		file1 = os.path.join(dir1,f) #源
		#判断是否是文件
		if os.path.isfile(file1):
			wj.append(os.path.getsize(file1))
		#判断是否是目录
		if os.path.isdir(file1):
			wj.append(calsize(file1)) #递归调用自己，来计算子文件夹大小
	s = 0
	for i in wj:
             s+=i
	return s
# 测试
print('测试目录的大小为：',calsize("./aa"),'字节')